package com.know.web;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.UUID;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import my.util.ResponseUtils;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadBase;
import org.apache.commons.fileupload.ProgressListener;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

import com.know.entity.Images;
import com.know.service.ImageService;
import com.kzw.bean.Record;
import com.kzw.json.Jackson;


/**
*多文件上传，文件限制， 文件后缀检测
*/ 
@WebServlet("/upload")
public class UploadHandleServlet extends HttpServlet {

    /**
	 * 
	 */
	private static final long serialVersionUID = 1L;
	private ImageService imageService=new ImageService();
	public void doGet(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
                //得到上传文件的保存目录，将上传的文件存放于WEB-INF目录下，不允许外界直接访问，保证上传文件的安全
                String savePath = this.getServletContext().getRealPath("/upload");
                //上传时生成的临时文件保存目录
                String tempPath = this.getServletContext().getRealPath("/WEB-INF/temp");
                File tmpFile = new File(tempPath);
             
                //得到调用文件上传处的类型，及所属ID 并设置
                String img_type=request.getParameter("imgType");
                String img_type_id=request.getParameter("imgId");
             
                if (!tmpFile.exists()) {
                    //创建临时目录
                    tmpFile.mkdir();
                }
                
                Map<String, Object> map = new HashMap<String, Object>();//前台所需数据存储
                List<Images> images=new ArrayList<Images>();//图片存储信息组

                //消息提示
                String message = "";
                try{
                    //使用Apache文件上传组件处理文件上传步骤：
                    //1、创建一个DiskFileItemFactory工厂
                    DiskFileItemFactory factory = new DiskFileItemFactory();
                    //设置工厂的缓冲区的大小，当上传的文件大小超过缓冲区的大小时，就会生成一个临时文件存放到指定的临时目录当中。
                    factory.setSizeThreshold(1024*100);//设置缓冲区的大小为100KB，如果不指定，那么缓冲区的大小默认是10KB
                    //设置上传时生成的临时文件的保存目录
                    factory.setRepository(tmpFile);
                    //2、创建一个文件上传解析器
                    ServletFileUpload upload = new ServletFileUpload(factory);
                    //监听文件上传进度
                    upload.setProgressListener(new ProgressListener(){
                        public void update(long pBytesRead, long pContentLength, int arg2) {
                          
                        	
                        }
                    });
                     //解决上传文件名的中文乱码
                    upload.setHeaderEncoding("UTF-8"); 
                    //3、判断提交上来的数据是否是上传表单的数据
                    if(!ServletFileUpload.isMultipartContent(request)){
                        //按照传统方式获取数据
                        return;
                    }
                    
                   
                    //4、使用ServletFileUpload解析器解析上传数据，解析结果返回的是一个List<FileItem>集合，每一个FileItem对应一个Form表单的输入项
                    List<FileItem> list = upload.parseRequest(request);
                    for(FileItem item : list){
                        //如果fileitem中封装的是普通输入项的数据
                        if(item.isFormField()){
                            String name = item.getFieldName();
                            //解决普通输入项的数据的中文乱码问题
                            String value = item.getString("UTF-8");
                            //value = new String(value.getBytes("iso8859-1"),"UTF-8");
                            System.out.println(name + "=" + value);
                        }else{//如果fileitem中封装的是上传文件
                            //得到上传的文件名称，
                            String filename = item.getName();
                            System.out.println(filename);
                            if(filename==null || filename.trim().equals("")){
                                continue;
                            }
                            //注意：不同的浏览器提交的文件名是不一样的，有些浏览器提交上来的文件名是带有路径的，如：  c:\a\b\1.txt，而有些只是单纯的文件名，如：1.txt
                            //处理获取到的上传文件的文件名的路径部分，只保留文件名部分
                            filename = filename.substring(filename.lastIndexOf("\\")+1);
                            //得到上传文件的扩展名
                            String fileExtName = filename.substring(filename.lastIndexOf("."));
                            //如果需要限制上传的文件类型，那么可以通过文件的扩展名来判断上传的文件类型是否合法
                            System.out.println("上传的文件的扩展名是："+fileExtName);
                            //获取item中的上传文件的输入流
                            InputStream in = item.getInputStream();
                            //得到文件保存的名称
                            
                            String saveFilename = makeFileName(filename);
                            //得到文件的保存目录
                            String dir=maketimePath(saveFilename,savePath);
                            String realSavePath = savePath+dir;
                            
                            //创建一个文件输出流
                            FileOutputStream out = new FileOutputStream(realSavePath + "/" + saveFilename);
                            String datapath="upload"+dir+"/"+saveFilename;
                            System.out.println("数据库上传路径："+datapath);
                    	
                            //创建一个缓冲区
                            byte buffer[] = new byte[1024];
                            //判断输入流中的数据是否已经读完的标识
                            int len = 0;
                            //循环将输入流读入到缓冲区当中，(len=in.read(buffer))>0就表示in里面还有数据
                            while((len=in.read(buffer))>0){
                                //使用FileOutputStream输出流将缓冲区的数据写入到指定的目录(savePath + "\\" + filename)当中
                                out.write(buffer, 0, len);
                            }
                            //关闭输入流
                            in.close();
                            //关闭输出流
                            out.close();
                            //删除处理文件上传时生成的临时文件
                            //item.delete();
                            //前台所需回显的JSON地址
                        	map.put("src",datapath);
                    		map.put("title", saveFilename);
                    		message = "文件上传成功！";
                    		//存储图片信息
                    		Images image=new Images();
                    		image.setImg_id(saveFilename.replaceAll(fileExtName,""));
                            image.setImg_type(img_type);
                            	System.out.println(img_type);
                            if(img_type_id!=null)
                            	image.setImg_type_id(Long.parseLong(img_type_id));
                            image.setImg_path(datapath);
                    		images.add(image);
                        }
                    }
                }catch (FileUploadBase.FileSizeLimitExceededException e) {
                    e.printStackTrace();
                    message="单个文件超限";
                }catch (FileUploadBase.SizeLimitExceededException e) {
                    e.printStackTrace();

                    message="文件总大小超限";
                }catch (Exception e) {
                    message= "文件上传失败！";
                    e.printStackTrace();
                }
                //写入数据库
                imageService.saveImg(images);
                //返回信息
                Record record=new Record(true, 0,message);
            	
        		//map.put("title", fname);
                record.setData(map);
                String json=Jackson.me().toJson(record);
        		ResponseUtils.renderJson(response, json);		
    }
    
    /**
    *UUID编号生成
    */ 
    private String makeFileName(String filename){  //2.jpg
        //为防止文件覆盖的现象发生，要为上传文件产生一个唯一的文件名
      //  return UUID.randomUUID().toString() + "_" + filename;
        String fileExtName = filename.substring(filename.lastIndexOf("."));
    	return UUID.randomUUID().toString().replaceAll("-","")+fileExtName;
    }
    

    /**
     * 时间目录夹
     */
    private String maketimePath(String filename, String savepath){
        Calendar now = Calendar.getInstance();  
        int year=now.get(Calendar.YEAR);
        int month=now.get(Calendar.MONTH+1);
        int day=now.get(Calendar.DATE);
        String dir =  "/" + year + "/" + month + "/" + day;  //upload\2\3  upload\3\5
        String diry=savepath+dir;
        //File既可以代表文件也可以代表目录
        File file = new File(diry);
        //如果目录不存在
        if(!file.exists()){
            //创建目录
            file.mkdirs();
        }
        return dir;
        
    }

    public void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {

        doGet(request, response);
    }
}